// (c) 2008 Kickstone Technology Ltd

#pragma once
#include "resource.h"       // main symbols
#include "DDButton.h"


#if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA)
#error "Single-threaded COM objects are not properly supported on Windows CE platform, such as the Windows Mobile platforms that do not include full DCOM support. Define _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA to force ATL to support creating single-thread COM object's and allow use of it's single-threaded COM object implementations. The threading model in your rgs file was set to 'Free' as that is the only threading model supported in non DCOM Windows CE platforms."
#endif


class CIEWindow;

class ATL_NO_VTABLE CButtonDemoButton :
	public CComObjectRootEx<CComSingleThreadModel>,
	public CComCoClass<CButtonDemoButton, &CLSID_ButtonDemoButton>,
	public IObjectWithSiteImpl<CButtonDemoButton>,
	public IDispatchImpl<IButtonDemoButton, &IID_IButtonDemoButton, &LIBID_DDButtonLib, /*wMajor =*/ 1, /*wMinor =*/ 0>,
	public IOleCommandTarget
{
public:
	CButtonDemoButton(){
		ATLTRACE("\n - CButtonDemoButton::CButtonDemoButton()");
	}

	~CButtonDemoButton(){
		ATLTRACE("\n - CButtonDemoButton::~CButtonDemoButton()");
	}

	DECLARE_REGISTRY_RESOURCEID(IDR_BUTTONDEMOBUTTON)

	DECLARE_NOT_AGGREGATABLE(CButtonDemoButton)

	BEGIN_COM_MAP(CButtonDemoButton)
		COM_INTERFACE_ENTRY(IButtonDemoButton)
		COM_INTERFACE_ENTRY(IDispatch)
		COM_INTERFACE_ENTRY(IObjectWithSite)
		COM_INTERFACE_ENTRY(IOleCommandTarget)
	END_COM_MAP()



	DECLARE_PROTECT_FINAL_CONSTRUCT()

	HRESULT FinalConstruct();
	void FinalRelease();

	// IOleCommandTarget functions
	STDMETHOD(QueryStatus)(const GUID* pguidCmdGroup, ULONG cCmds, OLECMD prgCmds[], OLECMDTEXT* pCmdText);
	STDMETHOD(Exec)(const GUID*, DWORD nCmdID, DWORD, VARIANTARG*, VARIANTARG* pvaOut);

	// IOleObjectWithSite Methods
	STDMETHOD(SetSite)(IUnknown *pUnkSite);


private:

	// track the handle of the main window
	HWND m_hWnd;
	void getHWND();

	CComQIPtr<IWebBrowser2, &IID_IWebBrowser2> m_spWebBrowser2;

	CIEWindow *m_ieWindow;

};

OBJECT_ENTRY_AUTO(__uuidof(ButtonDemoButton), CButtonDemoButton)
